# coding: utf-8

import os
import matplotlib.pyplot as plt
import numpy as np
import matplotlib

font = {"family": "Arial", "weight": "normal", "size": 14}
matplotlib.rc("font", **font)
LineWidth = 3


# ECCV 2024: WeConvene: Learned Image Compression with  Wavelet-Domain Convolution and Entropy Model
bpp = [0.163, 0.203, 0.309, 0.452, 0.637, 0.897]
msssim = [12.7813, 13.6297, 15.3087, 17.0491, 18.7330, 20.5285]
weconvene, = plt.plot(bpp, msssim, "k-o", linewidth=LineWidth, label='WeConvene')


# NOTE: 绘图
plt.legend(handles=[weconvene], loc=4)
plt.grid()
plt.xlabel('Bpp')
plt.ylabel('MS-SSIM')
plt.title('Kodak dataset')
plt.savefig('kodak_msssim.png', format='png', dpi=600, bbox_inches='tight')
